from random import choice


class RandomWalk():

    def __init__(self, num_points=5000):
        self.num_points = num_points

        self.x_values = [0]
        self.y_values = [0]

    def fill_walk(self):
        while len(self.x_values) < self.num_points:
            # x,y轴的防线和距离
            x_step = self.get_step()
            y_step = self.get_step()

            if x_step == 0 and y_step == 0:
                continue

            # 计算下一次的坐标
            next_x = self.x_values[-1] + x_step
            next_y = self.y_values[-1] + y_step

            self.x_values.append(next_x)
            self.y_values.append(next_y)

    def get_step(self):
        direct = choice([1, -1])
        distance = choice(range(0, 5))
        step = direct * distance
        return step

    def get_die_step(self):
        direct = choice([1])
        distance = choice(range(1, 7))
        step = direct * distance
        return step
